package com.java.springcloud.mybatisplus.injector;

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.java.springcloud.mybatisplus.injector.methods.TdBatchInsert;
import com.java.springcloud.mybatisplus.injector.methods.TdInsert;
import com.java.springcloud.mybatisplus.injector.methods.TdSelectList;
import com.java.springcloud.mybatisplus.injector.methods.TdSelectMaps;
import org.springframework.stereotype.Component;

import java.util.Arrays;
import java.util.List;

/**
 * @author Administrator
 */
@Component
public class TdEngineSqlInjector extends DefaultSqlInjector {

    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        List<AbstractMethod> list = Arrays.asList(
                new TdInsert(),
                new TdBatchInsert(),
                new TdSelectList(),
                new TdSelectMaps()
        );
        methodList.addAll(list);
        return methodList;
    }

}
